#!/usr/bin/env python
# coding: utf-8

"""
pwkyc.py
create by cuiyu 2025-06-04
说明：
 * 排污口异常检测测试代码

用法:
  pwkyc.py
  * 输入 f1 和 f2 两个图片路径，输出 0~1.0 的值，表示两个图片的相似度，==1 表示完全相同，==0 表示完全不同

更新:
2025-06-04:
"""

import logging
from skimage.metrics import structural_similarity as ssim  
import cv2  
from pathlib import Path


def main():
    f1 = Path("/aaby/temp/today/aaaa/0002.jpg")
    f2 = Path("/aaby/temp/today/aaaa/0003.jpg")

    # 读取图像并转为灰度  
    img1 = cv2.imread(str(f1), cv2.IMREAD_GRAYSCALE)  # 直接读为灰度图  
    img2 = cv2.imread(str(f2), cv2.IMREAD_GRAYSCALE)  

    # 计算SSIM  
    ssim_value = ssim(img1, img2, data_range=255)  # data_range指定像素范围  
    print(f"SSIM值 (scikit-image): {ssim_value:.4f}")  
    pass

if __name__ == "__main__":
    logging.basicConfig(level=logging.DEBUG) 
    main()